import os.path

from utils.sftp_clean_data import SftpCleanData

if __name__ == '__main__':
    test = SftpCleanData()
    basedir = r"这里传递要清洗数据的主文件夹"

    # 示例
    folders_to_process = [
        rf"{basedir}\huanggangsftp",
        rf"{basedir}\huangshisftp",
        rf"{basedir}\xiantaosftp",
        rf"{basedir}\shennongjiasftp",
        rf"{basedir}\shiyansftp",
        rf"{basedir}\xianningsftp",
        rf"{basedir}\enshisftp",
        rf"{basedir}\yichangsftp",
        rf"{basedir}\qianjiangsftp",
    ]

    for folder in folders_to_process:
        test.process_csv_files(folder)

